package com.kly.springbootinit.interceptor;

import com.kly.springbootinit.common.UserContext;
import com.kly.springbootinit.model.entity.User;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import static com.kly.springbootinit.constant.UserConstant.USER_LOGIN_STATE;

/**
 * @author admin
 * @date 2025/5/22
 */
public class LoginInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        // 判断是不是静态方法
        if("OPTIONS".equals(request.getMethod())) {
            return true;
        }
        // 获取user
        Object userObj = request.getSession().getAttribute(USER_LOGIN_STATE);
        if(userObj != null) {
            // 写入UserContext
            User user = (User) userObj;
            UserContext.setUser(user);
        }
        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
        HandlerInterceptor.super.postHandle(request, response, handler, modelAndView);
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        UserContext.remove();
    }
}
